package com.dragon.framework.aspect;

import com.dragon.common.constant.BasicConstant;
import com.dragon.common.utils.MdcUtils;
import com.dragon.common.utils.ServletUtils;
import lombok.extern.slf4j.Slf4j;
import org.aspectj.lang.ProceedingJoinPoint;
import org.aspectj.lang.annotation.Around;
import org.aspectj.lang.annotation.Aspect;
import org.aspectj.lang.annotation.Pointcut;
import org.springframework.stereotype.Component;
import org.springframework.web.context.request.ServletRequestAttributes;

import java.util.Objects;

/**
 * 全局追踪ID切面
 *
 * @author dragon
 * @version 1.0
 * @date 2024/7/16 16:42
 */
@Slf4j
@Aspect
@Component
public class TraceIdAspect {

    /**
     * 匹配 TraceId 注解
     * <b>注意</b>
     * 不推荐用 execution(* com.dragon.admin..*(..)) 对所有方法切面
     * 比如 方法调用：a -> b -> c
     * 对所有方法切面会导致执行3次
     * 1.如果方法执行后不删除 TraceId 会导致同一时间执行的定时任务会获得相同的 TraceId
     * 2.如果删除 TraceId 会导致方法调用链会生成不同的 TraceId
     */
    @Pointcut("@annotation(com.dragon.common.annotation.TraceId)")
    public void pointcutTraceId() {
    }

    /**
     * traceId环绕切面
     *
     * @param point 切面对象
     * @return 切面对象
     * @throws Throwable 异常
     */
    @Around("pointcutTraceId()")
    public Object traceIdAround(ProceedingJoinPoint point) throws Throwable {
        // 方法执行前
        ServletRequestAttributes requestAttributes = ServletUtils.getRequestAttributes();

        String traceId = MdcUtils.safeGenerate(BasicConstant.TRACE_ID);
        // 设置追踪ID
        MdcUtils.put(BasicConstant.TRACE_ID, traceId);

        // 执行方法
        Object result = point.proceed();

        // 方法执行后
        if (Objects.nonNull(requestAttributes)) {
            if (Objects.nonNull(requestAttributes.getResponse())) {
                // 设置响应头
                requestAttributes.getResponse().setHeader(BasicConstant.HEADER_TRACE_ID, traceId);
            }
        } else {
            // requestAttributes 为空则代表非http请求，避免和 TraceIdFilter 冲突删除 TRACE_ID
            MdcUtils.remove(BasicConstant.TRACE_ID);
        }

        return result;
    }

}
